#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import argparse

def args_parse():
    parser = argparse.ArgumentParser(prog='./caffe_proto_fix.py'
        , description='删除caffe prototxt文件中的指定Field(只处理Layer下的第一层Field)')

    parser.add_argument ('file'
        , type = str
        , help='指定要处理的文件')

    parser.add_argument ('-p'
        , '--proto'
        , type = str
        , help='指定要使用的caffe.proto, 确实使用caffe官方的caffe.proto')

    parser.add_argument ('-cf'
        , '--clear-field'
        , nargs = '*'
        , type = str
        , help='指定要清除的Field列表(不区分大小写)')

    parser.add_argument ('-lt'
        , '--layer-type'
        , type = str
        , help='指定要清除Field所在的层类型, 只有该类型下的指定field才被清除(不区分大小写)')

    parser.add_argument ('-o'
        , '--output'
        , required = True
        , type = str
        , help='指定输出文件')
    
    parser.add_argument('-v'
        , '--verbosity' 
        , action = 'count'
        , default = 0
        , help = '日志级别')

    return parser.parse_args()
